simple interest and compound interest in python